package com.zl.demo.web;

import com.zl.demo.pojo.User;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.MediaType;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;

import java.time.Duration;
import java.util.ArrayList;
import java.util.Random;
import java.util.concurrent.CompletableFuture;

@RestController
@Slf4j
@RequestMapping("user")
public class UserController {
    /**
     * 同步阻塞的handle
     * @return
     */
    @GetMapping("/sync")
    public User getUserSync() {
        log.info("sync 开始执行");
        User user = getUser();
        log.info("sync 执行完毕");
        return user;
    }

    /**
     * 异步的handle
     * @return
     */
    @GetMapping("/mono")
    public Mono<User> getUserMono() {
        log.info("mono 开始执行");
        Mono<User> mono = Mono.fromSupplier(this::getUser);
        log.info("mono 执行完毕");
        return mono;
    }

    /**
     *  数据全部组织为json数组一次性返回
     * @return
     */
    @GetMapping("/flux")
    public Flux<User> getUserFlux() {
        log.info("flux 开始执行");
        Flux<User> flux = getUserWithFlux();
        log.info("flux 执行完毕");
        return flux;
    }

    /**
     *  数据流水一样 ，有一点就返回给页面
     * @return
     */
    @GetMapping(value = "/stream", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
    public Flux<User> getUserStream() {
        log.info("stream 开始执行");
        Flux<User> flux = getUserWithFlux();
        log.info("stream 执行完毕");
        return flux;
    }

    /**
     * 创建flux的API
     * @return
     */

    public void testFlux() {
        //简单方法
            // 生成包含两个字符串的Flux："Hello", "World"
        Flux.just("SS", "BB").subscribe(System.out::println);
            //从一个Integer数组中创建Flux
        Flux.fromArray(new Integer[]{1, 2, 3}).subscribe(System.out::println);
            // 空的Flux
        Flux.empty().subscribe(System.out::println);
            // 创建一个包含1到10的整数Integer类型的Flux
        Flux.range(1, 10).subscribe(System.out::println);
            // 每隔1秒生成一个元素
        Flux.interval(Duration.ofSeconds(1)).subscribe(System.out::println);

        //generate方法
        Flux<String> flux = Flux.generate(sink -> {
            //添加元素
            sink.next("hello");
            //结束
            sink.complete();
        });
        //generate方法
            //定义一个随机生成器
        final Random random = new Random();
            // 把集合的构造函数作为初始化方法，形成的初始化变量，会传递给第二个函数作为参数
        Flux<Integer> flux1 = Flux.generate(ArrayList::new,(list,sink) ->{
            //生成随机数
            int value = random.nextInt(100);
            // 加入集合
            list.add(value);
            //加入Flux
            sink.next(value);
            if (list.size() == 10) {
                // 当集合元素数量等于10，则停止
                sink.complete();
            }
            return list;
        });

        //3.create方法
        Flux<Integer> flux2 = Flux.create(sink -> {
            for (int i = 0; i < 10; i++) {
                // 将循环遍历得到的数字，添加到flux
                sink.next(i);
            }
            //循环结束，任务完成
            sink.complete();
        });
        // 会生成一个包含从0到9的int元素的Flux
        flux.subscribe(System.out::println);

    }

    /**
     * 创建Mono的API
     * @return
     */
    public void testMono() {
        Mono<String> mono1 = Mono.fromSupplier(() -> "hello");
        Mono<String> mono2 = Mono.fromFuture(CompletableFuture.supplyAsync(() -> "hello"));
    }
    //生成etUserWithFlux()的方法
    private Flux<User> getUserWithFlux() {
        return Flux.interval(Duration.ofSeconds(1))// 每隔1秒发射一个元素
                .take(3) // 取前3个元素
                // 将元素转为一个User对象
                .map(i -> User.zl("user_" + i, 20 + i.intValue()));
    }

    // 生成user的方法
    private User getUser(){
        try {
            // 模拟业务耗时1秒
            Thread.sleep(3000);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        return User.zl("Rose", 18);
    }
}
